#构建提示词
from llama_index.core import PromptTemplate
from llama_index.core import ChatPromptTemplate
from llama_index.core.llms import ChatMessage,MessageRole

template = ("把语句 \"{text} \" 翻译成 {language}")
prompt_template = PromptTemplate(template=template)
prompt = prompt_template.format(text="今天天气怎么样？",language="日文")
print(prompt)
print("---------------------------------")

msg = prompt_template.format_messages(text="今天天气怎么样？",language="日文")
print(msg)
print("---------------------------------")

msg_template = [
    ChatMessage(content="你是一个专业的语言翻译，请将用户的语句翻译成{language}.",role=MessageRole.SYSTEM),
    ChatMessage(content="{text}",role=MessageRole.USER)
]
chat_template = ChatPromptTemplate(message_templates=msg_template)
prompt_1 = chat_template.format(text="今天天气怎么样？",language="日文")
print(prompt_1)
print("---------------------------------")

prompt_2 = chat_template.format_messages(text="今天天气怎么样？",language="日文")
print(prompt_2)
print("---------------------------------")